package leetcode.top100;

import utils.ListNode;

/**
 * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
 * <p>
 * 如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
 * <p>
 * 您可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * <p>
 * 示例：
 * <p>
 * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
 * 输出：7 -> 0 -> 8
 * 原因：342 + 465 = 807
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/add-two-numbers
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @date 2020/3/17 22:20
 */
public class Code02_TwoNumSumInList {
    // 代码太冗余，见修改后的代码
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        if(l1 == null || l2 == null) return  l1 == null ? l2 : l1;
        ListNode head = new ListNode(0);
        ListNode p1 = head;
        int carry = 0;
        int sum = 0;
        while (l1 != null && l2 != null){
            sum = l1.val + l2.val + carry;
            if(sum >= 10){
                sum -= 10;
                carry = 1;
            }else {
                carry = 0;
            }
            p1.next = new ListNode(sum);
            p1 = p1.next;
            l1 = l1.next;
            l2 = l2.next;
        }
        while (l1 != null){
            sum = l1.val +carry;
            if(sum >= 10){
                sum -= 10;
                carry = 1;
            }else {
                carry = 0;
            }
            p1.next = new ListNode(sum);
            p1 = p1.next;
            l1 = l1.next;
        }
        while (l2 != null){
            sum = l2.val +carry;
            if(sum >= 10){
                sum -= 10;
                carry = 1;
            }else {
                carry = 0;
            }
            p1.next = new ListNode(sum);
            p1 = p1.next;
            l2 = l2.next;
        }
        if(carry == 1){
            p1.next = new ListNode(1);
        }
        return head.next;
    }

    public ListNode addTwoNumbers2(ListNode l1, ListNode l2) {
        if(l1 == null || l2 == null) return  l1 == null ? l2 : l1;
        ListNode head = new ListNode(0);
        ListNode p1 = head;
        int carry = 0;
        int sum = 0;
        while (l1 != null || l2 != null){
            int val1 = l1 == null ? 0 : l1.val;
            int val2 = l2 == null ? 0 : l2.val;
            sum = val1 + val2 + carry;
            carry = sum / 10;
            p1.next = new ListNode(sum % 10);
            p1 = p1.next;
            l1 = l1 == null ? null : l1.next;
            l2 = l2 == null ? null : l2.next;
        }
        if(carry == 1){
            p1.next = new ListNode(1);
        }
        return head.next;
    }

    public static void main(String[] args) {
        ListNode l1 = new ListNode(1);
        l1.next = new ListNode(8);
        ListNode l2 = new ListNode(0);
        System.out.println(new Code02_TwoNumSumInList().addTwoNumbers2(l1, l2).val);
    }

}
